#pragma once

#include <iostream>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/fcntl.h>
#include <sys/wait.h>

#include "../comm/log.hpp"
#include "../comm/util.hpp"

// 负责形成编译文件，并编译

namespace ns_compiler
{
    using namespace ns_log;
    using namespace ns_util;
    class Compiler
    {
    public:
        Compiler()
        {
        }
        ~Compiler()
        {
        }

        // 只需要传临时文件的文件名，函数内部会根据文件名找到源文件
        // 编译成功时，会生成可执行程序
        static bool Compile(const std::string &filename)
        {
            // 创建子进程
            pid_t pid = fork();

            if (pid < 0)
            {
                // 创建失败
                LOG(FATAL) << " 内部错误，创建子进程失败 " << "\n";
                return false;
            }
            else if (pid == 0)
            {
                // 子进程

                // 创建子进程，防止程序替换影响程序的正常运行
                
                // 打开 .compile_error 文件
                umask(0);
                int _stderr = open(PathUtil::CompileError(filename).c_str(), O_CREAT | O_WRONLY, 0644);

                if (_stderr < 0)
                {
                    LOG(WARNING) << " 生成 stderr 文件失败 " << "\n";
                    exit(1);
                }

                // 把g++ 编译的错误信息输出到 .stderr 文件中（重定向），没有错误则文件内容为空
                dup2(_stderr, 2);

                // 程序替换，开始编译
                // g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(filename).c_str(), PathUtil::Src(filename).c_str(), "-D", "COMPILER_ONLINE","-std=c++11", nullptr);

                // 替换成功，则执行替换后的代码

                // 替换失败，继续执行子进程代码
                LOG(ERROR) << " 启动 g++ 失败，可能参数错误 " << "\n";
                exit(2);
            }
            else
            {
                // 父进程

                waitpid(pid, nullptr, 0); // 阻塞等待子进程的执行结果

                // 判断是否编译成功(编译成功会生成可执行程序)
                if (FileUtil::IsFileExists(PathUtil::Exe(filename)))
                {
                    LOG(INFO) << PathUtil::Src(filename) << " 编译成功，生成可执行程序 " << "\n";
                    return true;
                }
            }
            LOG(ERROR) << " 编译失败，未形成可执行程序 " << "\n";
            return false;
        }
    };
}